leetcode 412 python